#coding=utf-8
# 框架的驱动：
''' 框架驱动运行原理
    1.获取框架的位置
    2.读取配置文件，数据文件中的数据或配置信息
    3.读取测试脚本，运行脚本--调用公共函数库
    4.生成测试报告--错误截图
'''
import unittest, time, re
import HTMLTestRunner
import sys
reload(sys)
sys.setdefaultencoding('utf8')

# 创建测试集 create_suite()
# 获取框架的位置，寻找test*.py文件
def create_suite():
    testunit = unittest.TestSuite()
    # 获取脚本位置
    test_dir='F:\\PyWorkSpace\\iwebshop_framework\\test_case'
    # 调用discover()
    discv = unittest.defaultTestLoader.discover(test_dir,pattern='test_B*.py')
    # 从discover中寻找测试用例
    for test_case in discv:
        print test_case
        testunit.addTest(test_case)
    return testunit

# 测试报告
now = time.strftime('%Y-%m-%d %H-%M-%S')
# 定义报告存放路径
filename="F:\\PyWorkSpace\\iwebshop_framework\\report\\"+now+"result.html"
fo=open(filename,'wb')
runner=HTMLTestRunner.HTMLTestRunner(stream=fo,
                              title='iwebshop测试报告',
                              description='用例执行情况:')

if __name__=='__main__':
    testnames=create_suite()
    runner.run(testnames)
    fo.close()
